def square(x):
    """
    计算平方并返回结果
    >>> square(2)
    4
    >>> square(3)
    9
    """
    return x ** x


def product(x, y):
    pass

if __name__ == '__main__':
    import doctest
    import my_math
    doctest.testmod(my_math)
